# 定义判断主板函数（优化版）
def is_main_board(stock_code):
    """判断股票是否属于沪深主板"""
    code = str(stock_code).strip()
    if len(code) < 3:
        return False
    prefix = code[:3]
    # 沪市主板: 60开头；深市主板: 000/001开头
    return prefix.startswith('60') or prefix.startswith('000') or prefix.startswith('001')